#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <arpa/inet.h>

int main(int argc, char **argv) {
    if (argc != 2) {
        printf("用法: %s <主机名>\n", argv[0]);
        return 1;
    }

    const char *hostname = argv[1];
    struct addrinfo hints, *res, *p;
    int status;
    char ipstr[INET6_ADDRSTRLEN];

    // 初始化 hints 结构体
    memset(&hints, 0, sizeof hints);
    hints.ai_family = AF_UNSPEC;    // AF_INET 或 AF_INET6 来强制使用特定版本
    hints.ai_socktype = SOCK_STREAM; // TCP 流

    // 调用 getaddrinfo 函数来解析主机名
    if ((status = getaddrinfo(hostname, NULL, &hints, &res)) != 0) {
        fprintf(stderr, "getaddrinfo 错误: %s\n", gai_strerror(status));
        return 2;
    }

    printf("%s 的 IP 地址:\n\n", hostname);

    // 遍历返回的结果链表
    for (p = res; p != NULL; p = p->ai_next) {
        void *addr;
        const char *ipver;

        // 获取指向地址的指针，区分 IPv4 和 IPv6
        if (p->ai_family == AF_INET) { // IPv4
            struct sockaddr_in *ipv4 = (struct sockaddr_in *)p->ai_addr;
            addr = &(ipv4->sin_addr);
            ipver = "IPv4";
        } else { // IPv6
            struct sockaddr_in6 *ipv6 = (struct sockaddr_in6 *)p->ai_addr;
            addr = &(ipv6->sin6_addr);
            ipver = "IPv6";
        }

        // 将 IP 地址转换为字符串并打印
        inet_ntop(p->ai_family, addr, ipstr, sizeof ipstr);
        printf("  %s: %s\n", ipver, ipstr);
    }

    // 释放 getaddrinfo 分配的内存
    freeaddrinfo(res);

    return 0;
}
